#  -*- coding: utf-8 -*-

from strategy.stock_pool.low_pe_stock_pool import LowPeStockPool
from strategy.stock_pool.trending_stock_pool import TrendingStockPool
from strategy.stock_pool.trending_speedup_stock_pool import TrendingSpeedUpStockPool
from strategy.stock_pool.short_strength_stock_pool import ShortStrengthStockPool
from strategy.stock_pool.manual_stock_pool import  ManualStockPool

class StockPoolFactory:
    @staticmethod
    def get_stock_pool(name, strategy_name,begin_date, end_date, interval):
        if name == 'low_pe_stock_pool':
            print("excute")
            return LowPeStockPool(strategy_name,begin_date, end_date, interval)
        if name == 'trending_stock_pool':
            print("excute")
            return TrendingStockPool(strategy_name,begin_date, end_date, interval)

        if name == 'trending_speedup_stock_pool':
            print("excute")
            return TrendingSpeedUpStockPool(strategy_name,begin_date, end_date, interval)

        if name == 'short_strength_stock_pool':
            print("excute")
            return ShortStrengthStockPool(strategy_name,begin_date, end_date, interval)

        if name == 'manual_stock_pool':
            print("excute")
            return ManualStockPool(strategy_name,begin_date, end_date, interval)